import os
import cv2
import numpy as np
import time
import matplotlib.pyplot as plt

#目前只写了两个函数 一个是读取图片 一个是检测文件夹  后面仍需要补充显示图片  保存图片  从一个文件夹复制图片到另外一个文件夹

#读取图片
def read_pic(path,fileName):
    """读取图片，输入的是一个图片所在的路径 已经图片的名称 都是str格式"""
    imgpath=os.path.join(path,fileName)
    return cv2.imdecode(np.fromfile(imgpath, dtype=np.uint16), -1)

#保存图片
def save_pic(pic_img,path,fileName):
    """输入的是图片的矩阵，路径，还有图片的名字"""
    savepath=os.path.join(path,fileName)
    cv2.imencode('.tif', pic_img)[1].tofile(savepath)

#检测文件夹
def check_dir(path):
    """检测文件夹是否寸在，如果存在就打印已经存在 不用新建  如果不存在 就新建一个"""
    if os.path.exists(path):
        print("文件夹： ",path,"已经寸在")
        time.sleep(0.5)
    else:
        print("文件夹： ", path, "不寸在",end='')
        os.makedirs(path)
        print("已新建")
        time.sleep(3)

#显示图片
def show_pic(img,title):
    """输入这个图片的矩阵 然后显示出来 标题是title"""
    # pass
    fig, ax = plt.subplots()
    # ax.axis('off')
    ax.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
    plt.title(title)
    plt.show()

